package com.bosicen.crm.tools;

/**
 *  缩略图实现，将图片(jpg、bmp、png、gif等等)真实的变成想要的大小
 */

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import javax.imageio.ImageIO;

import org.springframework.web.multipart.MultipartFile;

import com.sun.image.codec.jpeg.JPEGCodec;
import com.sun.image.codec.jpeg.JPEGImageEncoder;

/*******************************************************************************
 * 缩略图类（通用） 本java类能将jpg、bmp、png、gif图片文件，进行等比或非等比的大小转换。 具体使用方法
 * compressPic(大图片路径,生成小图片路径,大图片文件名,生成小图片文名,生成小图片宽度,生成小图片高度,是否等比缩放(默认为true))
 */
public class CompressPic {

	/*
	 * 获得图片大小 传入参数 String path ：图片路径
	 */
	public long getPicSize(String path) {
		file = new File(path);
		return file.length();
	}

	/**
	 * 图片处理
	 * 
	 * @return
	 */
	public boolean compressPic() {
		FileOutputStream out = null;
		int newWidth, newHeight;
		try {
			// 获得源文件
			file = new File(inputDir + inputFileName);

			// 如果目录不存在，新建
			fileExists(inputDir);
			Image img = ImageIO.read(file);
			// 判断图片格式是否正确
			if (img.getWidth(null) == -1) {
				System.out.println("不能读取文件，请重试");
				return false;
			} else {
				// 判断是否是等比缩放
				if (this.proportion == true) {
					// 为等比缩放计算输出的图片宽度及高度
					double rate1 = (AppUtil.ObjectToDouble(img.getWidth(null))) / (AppUtil.ObjectToDouble(outputWidth) + 0.1);
					double rate2 = (AppUtil.ObjectToDouble(img.getHeight(null))) / (AppUtil.ObjectToDouble(outputHeight) + 0.1);
					// 根据缩放比率大的进行缩放控制
					double rate = rate1 > rate2 ? rate1 : rate2;
					newWidth = AppUtil.ObjectToInt(Math.round(((AppUtil.ObjectToDouble(img.getWidth(null))) / rate)));
					newHeight = AppUtil.ObjectToInt(Math.round(((AppUtil.ObjectToDouble(img.getHeight(null))) / rate)));
				} else {
					newWidth = outputWidth; // 输出的图片宽度
					newHeight = outputHeight; // 输出的图片高度
				}

				/*
				 * Image.SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
				 */
				BufferedImage tag = new BufferedImage(newWidth, newHeight, BufferedImage.TYPE_INT_RGB);
				tag.getGraphics().drawImage(img.getScaledInstance(newWidth, newHeight, Image.SCALE_SMOOTH), 0, 0, null);
				out = new FileOutputStream(outputDir + outputFileName);
				// JPEGImageEncoder可适用于其他图片类型的转换
				JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
				encoder.encode(tag);

				// 宽度大于120加水印
//				if (newWidth > 120) {
//					pressImage("", outputDir + outputFileName, 20, 20);
//				}
			}
		} catch (IOException ex) {
			ex.printStackTrace();
		} finally {
			try {
				if (out != null)
					out.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return false;
	}

	/**
	 * 图片处理
	 * 
	 * @return
	 */
	public boolean compressPic(byte[] fileTobytes) {
		FileOutputStream out = null;
		int newWidth, newHeight;
		try {
			String tpath = inputDir + inputFileName;
			// 获得源文件
			file = new File(tpath);
			fileExists(inputDir);

			writeFile(tpath, fileTobytes);
			Image img = ImageIO.read(file);
			if (img.getWidth(null) == -1) {
				System.out.println("文件不正确，请重试");
				return false;
			} else {
				// 判断是否是等比缩放
				if (this.proportion == true) {
					// 为等比缩放计算输出的图片宽度及高度
					double rate1 = (AppUtil.ObjectToDouble(img.getWidth(null))) / (AppUtil.ObjectToDouble(outputWidth) + 0.1);
					double rate2 = (AppUtil.ObjectToDouble(img.getHeight(null))) / (AppUtil.ObjectToDouble(outputHeight) + 0.1);
					// 根据缩放比率大的进行缩放控制
					double rate = rate1 > rate2 ? rate1 : rate2;
					newWidth = AppUtil.ObjectToInt(Math.round(((AppUtil.ObjectToDouble(img.getWidth(null))) / rate)));
					newHeight = AppUtil.ObjectToInt(Math.round(((AppUtil.ObjectToDouble(img.getHeight(null))) / rate)));
				} else {
					newWidth = outputWidth; // 输出的图片宽度
					newHeight = outputHeight; // 输出的图片高度
				}

				/*
				 * Image.SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
				 */
				BufferedImage tag = new BufferedImage(newWidth, newHeight, BufferedImage.TYPE_INT_RGB);
				tag.getGraphics().drawImage(img.getScaledInstance(newWidth, newHeight, Image.SCALE_SMOOTH), 0, 0, null);
				out = new FileOutputStream(outputDir + outputFileName);
				// JPEGImageEncoder可适用于其他图片类型的转换
				JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
				encoder.encode(tag);

				// 宽度大于120加水印
//				if (newWidth > 120) {
//					pressImage("", outputDir + outputFileName, 20, 20);
//				}
			}
		} catch (IOException ex) {
			ex.printStackTrace();
		} finally {
			try {
				if (out != null)
					out.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return true;
	}

	/**
	 * 写文件
	 * 
	 * @param path
	 * @param filebytes
	 */
	public static void writeFile(String path, byte[] filebytes) {
		FileOutputStream fos = null;
		try {
			fos = new FileOutputStream(path); // 写入文件
			fos.write(filebytes);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				fos.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

	/**
	 *如果目录不存在，新建
	 * @param filePath
	 */
	public void fileExists(String filePath){
		File file = new File(filePath);
		if (!file.exists()) {
			file.mkdirs();
		}
	}
	/**
	 * 切图
	 * 
	 * @param inputDir
	 *            输入图路径
	 * @param outputDir
	 *            输出图路径
	 * @param inputFileName
	 *            输入图文件名
	 * @param outputFileName
	 *            输出图文件名
	 * @return
	 */
	public boolean compressPic(String inputDir, String outputDir, String inputFileName, String outputFileName) {
		this.inputDir = inputDir;
		this.outputDir = outputDir;
		this.inputFileName = inputFileName;
		this.outputFileName = outputFileName;
		return compressPic();
	}

	/**
	 * 按自定义长宽是否等比压缩切图
	 * 
	 * @param inputDir
	 *            输入图路径
	 * @param outputDir
	 *            输出图路径
	 * @param inputFileName
	 *            输入图文件名
	 * @param outputFileName
	 *            输出图文件名
	 * @param width
	 *            设置图片长宽
	 * @param height
	 * @param gp
	 *            是否是等比缩放 标记
	 * @return
	 */
	public boolean compressPic(String inputDir, String outputDir, String inputFileName, String outputFileName, int width, int height,
			boolean gp) {
		this.inputDir = inputDir;
		this.outputDir = outputDir;
		this.inputFileName = inputFileName;
		this.outputFileName = outputFileName;
		setWidthAndHeight(width, height);
		this.proportion = gp;
		return compressPic();
	}

	/**
	 * 按自定义长宽是否等比压缩切图
	 * 
	 * @param inputDir
	 *            输入图路径
	 * @param outputDir
	 *            输出图路径
	 * @param inputFileName
	 *            输入图文件名
	 * @param outputFileName
	 *            输出图文件名
	 * @param width
	 *            设置图片长宽
	 * @param height
	 * @param gp
	 *            是否是等比缩放 标记
	 * @param file 文件
	 * @return
	 */
	public boolean compressPic(String inputDir, String outputDir, String inputFileName, String outputFileName, int width, int height,
			boolean gp, byte[] file) {
		this.inputDir = inputDir;
		this.outputDir = outputDir;
		this.inputFileName = inputFileName;
		this.outputFileName = outputFileName;
		setWidthAndHeight(width, height);
		this.proportion = gp;
		return compressPic(file);
	}

	/**
	 * 把图片印刷到图片上
	 * 
	 * @param pressImg
	 *            水印文件
	 * @param targetImg
	 *            目标文件
	 * @param x
	 * @param y
	 */
	public final static void pressImage(String pressImg, String targetImg, int x, int y) {
		FileOutputStream out = null;
		try {
			File _file = new File(targetImg);
			Image src = ImageIO.read(_file);
			int wideth = src.getWidth(null);
			int height = src.getHeight(null);
			BufferedImage image = new BufferedImage(wideth, height, BufferedImage.TYPE_INT_RGB);
			Graphics g = image.createGraphics();
			g.drawImage(src, 0, 0, wideth, height, null);

			// 水印文件
			File _filebiao = new File(pressImg);
			Image src_biao = ImageIO.read(_filebiao);
			int wideth_biao = src_biao.getWidth(null);
			int height_biao = src_biao.getHeight(null);
			g.drawImage(src_biao, wideth - wideth_biao - x, height - height_biao - y, wideth_biao, height_biao, null);
			// /
			g.dispose();
			out = new FileOutputStream(targetImg);
			JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
			encoder.encode(image);
			out.close();
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				if (out != null)
					out.close();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}

	/**
	 * 打印文字水印图片
	 * 
	 * @param pressText
	 *            文字
	 * @param targetImg
	 *            目标图片
	 * @param fontName
	 *            字体名
	 * @param fontStyle
	 *            字体样式
	 * @param color
	 *            字体颜色
	 * @param fontSize
	 *            字体大小
	 * @param x
	 *            偏移量
	 * @param y
	 */

	public static void pressText(String pressText, String targetImg, String fontName, int fontStyle, int color, int fontSize, int x, int y) {
		FileOutputStream out = null;
		try {
			File _file = new File(targetImg);
			Image src = ImageIO.read(_file);
			int wideth = src.getWidth(null);
			int height = src.getHeight(null);
			BufferedImage image = new BufferedImage(wideth, height, BufferedImage.TYPE_INT_RGB);
			Graphics g = image.createGraphics();
			g.drawImage(src, 0, 0, wideth, height, null);
			// String s="www.qhd.com.cn";
			g.setColor(Color.RED);
			g.setFont(new Font(fontName, fontStyle, fontSize));

			g.drawString(pressText, wideth - fontSize - x, height - fontSize / 2 - y);
			g.dispose();
			out = new FileOutputStream(targetImg);
			JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
			encoder.encode(image);
		} catch (Exception e) {
			System.out.println(e);
		} finally {
			try {
				if (out != null)
					out.close();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}

	/**
	 * 不按比例保存图片文件
	 * @param file MultipartFile
	 * @param path 保存目录
	 * @return
	 * @throws Exception
	 */
	public static boolean noProportionalSaveFile(MultipartFile file,String path,String filename) throws Exception {
		boolean flag = false;
		CompressPic cp = new CompressPic();
		if (file == null) {
			throw new Exception("上传失败：文件为空");
		}
		if (file.getSize() > 10000000) {
			throw new Exception("上传失败：文件大小不能超过10M");
		}
		if (file.getSize() > 0) {
			// 保存原图及120 X 90的缩略图
			flag = cp.compressPic(path, path, filename + "_3.jpg", filename + "_1.jpg", 120, 90, false, file.getBytes());

			// 保存320 X 240的缩略图
			flag = cp.compressPic(path, path, filename + "_3.jpg", filename + "_2.jpg", 320, 240, false);
		} else {
			throw new Exception("上传失败：上传文件不能为空");
		}
		return flag;
	}
	
	/**
	 * 按比例保存图片文件
	 * @param file MultipartFile
	 * @param path 保存目录
	 * @return
	 * @throws Exception
	 */
	public static boolean proportionalSaveFile(MultipartFile file,String path,String filename) throws Exception {
		boolean flag = false;
		CompressPic cp = new CompressPic();
		if (file == null) {
			throw new Exception("上传失败：文件为空");
		}
		if (file.getSize() > 10000000) {
			throw new Exception("上传失败：文件大小不能超过10M");
		}
		if (file.getSize() > 0) {
			// 保存原图及120 X 90的缩略图
			flag = cp.compressPic(path, path, filename + "_3.jpg", filename + "_1.jpg", 120, 90, true, file.getBytes());

			// 保存320 X 240的缩略图
			flag = cp.compressPic(path, path, filename + "_3.jpg", filename + "_2.jpg", 320, 240, true);
		} else {
			throw new Exception("上传失败：上传文件不能为空");
		}
		return flag;
	}
	
	// main测试Proportional
	// compressPic(大图片路径,生成小图片路径,大图片文件名,生成小图片文名,生成小图片宽度,生成小图片高度,是否等比缩放(默认为true))
	public static void main(String[] arg) {
		CompressPic mypic = new CompressPic();
		System.out.println("输入的图片大小：" + mypic.getPicSize("d:\\testimage\\t1.jpg") / 1024 + "KB");
		int count = 0; // 记录全部图片压缩所用时间

		for (int i = 0; i < 10; i++) {
			int start = (int) System.currentTimeMillis(); // 开始时间
			mypic.compressPic("D:\\testimage\\", "d:\\cutimage\\", "t1.jpg", "r1" + i + ".jpg", 120, 90, true);
			int end = (int) System.currentTimeMillis(); // 结束时间
			int re = end - start; // 但图片生成处理时间
			count += re;
			System.out.println("第" + (i + 1) + "张图片压缩处理使用了: " + re + "毫秒");
			System.out.println("输出的图片大小：" + mypic.getPicSize("e:\\cupimage\\r1" + i + ".jpg") / 1024 + "KB");
		}
		System.out.println("总共用了：" + count + "毫秒");
	}

	private File file = null; // 文件对象
	private String inputDir; // 输入图路径
	private String outputDir; // 输出图路径
	private String inputFileName; // 输入图文件名
	private String outputFileName; // 输出图文件名
	private int outputWidth = 100; // 默认输出图片宽
	private int outputHeight = 100; // 默认输出图片高
	private boolean proportion = true; // 是否等比缩放标记(默认为等比缩放)

	public CompressPic() { // 初始化变量
		inputDir = "";
		outputDir = "";
		inputFileName = "";
		outputFileName = "";
		outputWidth = 100;
		outputHeight = 100;
	}

	public void setInputDir(String inputDir) {
		this.inputDir = inputDir;
	}

	public void setOutputDir(String outputDir) {
		this.outputDir = outputDir;
	}

	public void setInputFileName(String inputFileName) {
		this.inputFileName = inputFileName;
	}

	public void setOutputFileName(String outputFileName) {
		this.outputFileName = outputFileName;
	}

	public void setOutputWidth(int outputWidth) {
		this.outputWidth = outputWidth;
	}

	public void setOutputHeight(int outputHeight) {
		this.outputHeight = outputHeight;
	}

	public void setWidthAndHeight(int width, int height) {
		this.outputWidth = width;
		this.outputHeight = height;
	}
}
